/*
 * @(#)ASCellViewFactory.java   09/09/13
 *
 * Copyright (C) 2009 Ecole des Mines d'Ales
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Contact: yulinz88@gmail.com
 *
 *
 *
 */



package fr.ema.arch3d.gea.ass;

import fr.ema.arch3d.gea.port.MyPort;
import fr.ema.arch3d.gea.port.PortViewClient;
import fr.ema.arch3d.gea.port.PortViewControl;
import fr.ema.arch3d.gea.port.PortViewServer;

import org.jgraph.graph.DefaultCellViewFactory;
import org.jgraph.graph.DefaultGraphCell;
import org.jgraph.graph.Port;
import org.jgraph.graph.PortView;
import org.jgraph.graph.VertexView;

import java.util.Map;

/**
 * A default view factory for a JGraph. This simple factory associate a given
 * cell class to a cell view. This is a javabean, just parameter it correctly in
 * order it meets your requirements (else subclass it or subclass
 * DefaultCellViewFactory). You can also recover the gpConfiguration of that
 * javabean via an XML file via XMLEncoder/XMLDecoder.
 *
 */
public class ASCellViewFactory extends DefaultCellViewFactory {

    /** Field description */
    public static final String VIEW_CLASS_KEY = "viewClassKey";

    public static final void setViewClass(Map map, String viewClass) {
        map.put(VIEW_CLASS_KEY, viewClass);
    }

    protected VertexView createVertexView(Object v) {
        try {
            DefaultGraphCell cell      = (DefaultGraphCell) v;
            String           viewClass = (String) cell.getAttributes().get(VIEW_CLASS_KEY);;
            VertexView       view      =
                (VertexView) Thread.currentThread().getContextClassLoader().loadClass(viewClass).newInstance();

            view.setCell(v);

            return view;
        } catch (Exception ex) {}

        return super.createVertexView(v);
    }

    protected PortView createPortView(Port p) {
        if (((MyPort) p).portType.equals("server")) {
            return new PortViewServer((MyPort) p);
        } else if (((MyPort) p).portType.equals("client")) {
            return new PortViewClient((MyPort) p);
        } else if (((MyPort) p).portType.equals("controller")) {
            return new PortViewControl((MyPort) p);
        }

        return null;
    }
}
